package Leetcode;

import java.util.ArrayList;
import java.util.List;

/**
 * 给你一个 m 行 n 列的矩阵 matrix ，请按照 顺时针螺旋顺序 ，返回矩阵中的所有元素。
 * 输入：matrix = [[1,2,3],[4,5,6],[7,8,9]]
 * 输出：[1,2,3,6,9,8,7,4,5]
 */
public class 力扣54_螺旋矩阵 {
    public static void main(String[] args) {
        int[][] matrix0 = {{1, 2}, {3, 4}};
        System.out.println(spiralOrder(matrix0));
        int[][] matrix1 = {{1, 2, 3}, {4, 5, 6}, {7, 8, 9}};
        System.out.println(spiralOrder(matrix1));
        int[][] matrix2 = {{1, 2, 3, 4}, {5, 6, 7, 8}, {9, 10, 11, 12}};
        System.out.println(spiralOrder(matrix2));

        System.out.println(spiralOrder1(matrix0));
        System.out.println(spiralOrder1(matrix1));
        System.out.println(spiralOrder1(matrix2));
    }

    // 定义四个方向 每次循环判断是否到达边界条件 然后转向 并修改边界值
    static List<Integer> spiralOrder(int[][] matrix) {
        List<Integer> result = new ArrayList<>();
        boolean xAdd = true, xReduce = false, yAdd = false, yReduce = false;
        int xMin = 0, xMax = matrix[0].length;
        int yMin = 0, yMax = matrix.length;
        int x = 0, y = 0;
        while (x >= xMin && x < xMax && y >= yMin && y < yMax) {
            result.add(matrix[y][x]);
            if (xAdd) {
                if (x + 1 >= xMax) {
                    xAdd = false;
                    yAdd = true;
                    yMin++;
                    y++;
                    continue;
                }
                x++;
            } else if (yAdd) {
                if (y + 1 >= yMax) {
                    yAdd = false;
                    xReduce = true;
                    xMax--;
                    x--;
                    continue;
                }
                y++;
            } else if (xReduce) {
                if (x - 1 < xMin) {
                    xReduce = false;
                    yReduce = true;
                    yMax--;
                    y--;
                    continue;
                }
                x--;
            } else if (yReduce) {
                if (y - 1 < yMin) {
                    yReduce = false;
                    xAdd = true;
                    xMin++;
                    x++;
                    continue;
                }
                y--;
            }
        }
        return result;
    }

    // 每一轮 按照 上 右 下 左 转圈输出一行或一列数据 并修改边界值
    static List<Integer> spiralOrder1(int[][] matrix) {
        List<Integer> result = new ArrayList<>();
        int left = 0, right = matrix[0].length - 1;
        int top = 0, bottom = matrix.length - 1;
        // 总数
        int needNum = matrix.length * matrix[0].length;
        while (needNum > 0) {
            // 上
            for (int i = left; i <= right && needNum > 0; i++, needNum--) {
                result.add(matrix[top][i]);
            }
            top++;

            // 右
            for (int i = top; i <= bottom && needNum > 0; i++, needNum--) {
                result.add(matrix[i][right]);
            }
            right--;

            // 下
            for (int i = right; i >= left && needNum > 0; i--, needNum--) {
                result.add(matrix[bottom][i]);
            }
            bottom--;

            //左
            for (int i = bottom; i >= top && needNum > 0; i--, needNum--) {
                result.add(matrix[i][left]);
            }
            left++;
        }
        return result;
    }
}
